/* Code Pulse: a real-time code coverage tool, for more information, see <http://code-pulse.com/>
 *
 * Copyright (C) 2014-2017 Code Dx, Inc. <https://codedx.com/>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.secdec.bytefrog.agent.util

import java.io.ByteArrayOutputStream

import org.scalamock.scalatest.MockFactory
import com.codedx.codepulse.agent.message.BufferService
import com.codedx.codepulse.agent.message.MessageDealer
import com.codedx.codepulse.agent.common.message.MessageProtocol
import com.codedx.codepulse.agent.common.queue.BufferPool
import com.codedx.bytefrog.instrumentation.id._
import com.codedx.codepulse.agent.common.queue.DataBufferOutputStream

trait MockHelpers { self: MockFactory =>

	class MockBufferService extends BufferService  {
		protected def innerObtain(): DataBufferOutputStream = new DataBufferOutputStream(new ByteArrayOutputStream())
		protected def innerSend(buffer: DataBufferOutputStream): Unit = {}
	}

	class MessageDealerMockable extends MessageDealer(mock[MessageProtocol], mock[MockBufferService], mock[ClassIdentifier], mock[MethodIdentifier])

	class BufferPoolMockable extends BufferPool(10, 100)
}